package com.landinfo.web.rest.util;

import java.util.Arrays;
import java.util.Iterator;

/**
 * 检查上传文件格式工具
 */
public class CheckUploadFileUtil {

    // 文件允许格式
    private static String[] allowFiles = {".rar", ".doc", ".docx", ".zip", ".pdf", ".txt", ".swf", ".wmv", ".gif", ".png",
        ".jpg", ".jpeg", ".bmp", ".svg", ".icon", ".webp", ".mov", ".m4v", ".mp4", ".avi", ".mkv",  ".flv", ".mp3",
        ".csv", ".gpkg", ".xls", ".xlsx", ".ppt", ".pptx"};

    /**
     * 检查通用上传文件格式是否符合
     * @param fileName
     * @return
     */
    public static boolean checkFileType(String fileName) {
        Iterator<String> type = Arrays.asList(allowFiles).iterator();
        while (type.hasNext()) {
            String ext = type.next();
            if (fileName.toLowerCase().endsWith(ext)) {
                return true;
            }
        }
        return false;
    }

}
